package leetcode.dynamic;

import java.util.Arrays;

/**
 * 一条包含字母 A-Z 的消息通过以下映射进行了 编码 ：
 * <p>
 * 'A' -> "1"
 * 'B' -> "2"
 * ...
 * 'Z' -> "26"
 * 要 解码 已编码的消息，所有数字必须基于上述映射的方法，反向映射回字母（可能有多种方法）。例如，"11106" 可以映射为：
 * <p>
 * "AAJF" ，将消息分组为 (1 1 10 6)
 * "KJF" ，将消息分组为 (11 10 6)
 * 注意，消息不能分组为  (1 11 06) ，因为 "06" 不能映射为 "F" ，这是由于 "6" 和 "06" 在映射中并不等价。
 * <p>
 * 给你一个只含数字的 非空 字符串 s ，请计算并返回 解码 方法的 总数 。
 * <p>
 * 题目数据保证答案肯定是一个 32 位 的整数。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：s = "12"
 * 输出：2
 * 解释：它可以解码为 "AB"（1 2）或者 "L"（12）。
 * 示例 2：
 * <p>
 * 输入：s = "226"
 * 输出：3
 * 解释：它可以解码为 "BZ" (2 26), "VF" (22 6), 或者 "BBF" (2 2 6) 。
 * 示例 3：
 * <p>
 * 输入：s = "06"
 * 输出：0
 * 解释："06" 无法映射到 "F" ，因为存在前导零（"6" 和 "06" 并不等价）。
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= s.length <= 100
 * s 只包含数字，并且可能包含前导零。
 */
public class LeetCode91_NumDecodings2 {
    public int numDecodings(String s) {
        return f(s.toCharArray(), 0);
    }

    public int f(char[] arr, int i) {
        if (i >= arr.length) {
            return 1;
        }
        int ans = 0;
        if (arr[i] == '0') {
            return 0;
        }

        ans += f(arr, i + 1);
        if (i + 1 < arr.length && ((arr[i] - '0') * 10 + (arr[i + 1] - '0')) <= 26) {
            ans += f(arr, i + 2);
        }
        return ans;
    }

    public int numDecodings2(String s) {
        int[] dp = new int[s.length()];
        Arrays.fill(dp, -1);
        return f2(s.toCharArray(), 0, dp);
    }

    public int f2(char[] arr, int i, int[] dp) {
        if (i >= arr.length) {
            return 1;
        }

        if (dp[i] != -1) {
            return dp[i];
        }

        int ans = 0;
        if (arr[i] == '0') {
            return 0;
        }

        ans += f(arr, i + 1);
        if (i + 1 < arr.length && ((arr[i] - '0') * 10 + (arr[i + 1] - '0')) <= 26) {
            ans += f(arr, i + 2);
        }

        dp[i] = ans;

        return dp[0];
    }

    public int numDecodings3(String s) {
        char[] arr = s.toCharArray();
        int[] dp = new int[s.length() + 1];
        Arrays.fill(dp, 0);
        dp[s.length()] = 1;

        for (int i = arr.length - 1; i >= 0; i--) {
            int ans = 0;
            if (arr[i] == '0') {
                ans = 0;
            } else {
                ans += dp[i + 1];
                if (i + 1 < arr.length && ((arr[i] - '0') * 10 + (arr[i + 1] - '0')) <= 26) {
                    ans += dp[i + 2];
                }
            }
            dp[i] = ans;
        }

        return dp[0];
    }

    public static void main(String[] args) {
        LeetCode91_NumDecodings2 leetCode91_numDecodings2 = new LeetCode91_NumDecodings2();
//        int i = leetCode91_numDecodings2.numDecodings("12");
        int i = leetCode91_numDecodings2.numDecodings2("12");
        System.out.println("i = " + i);
    }
}
